import websocket

class WebSocketClient:
    def __init__(self, uri):
        self.uri = uri
        self.ws = websocket.WebSocket()

    def connect(self):
        self.ws.connect(self.uri)

    def send(self, message):
        self.ws.send(message)

    def receive(self):
        return self.ws.recv()